'''
下载所有股票信息，包括4列：序号、编号、是否有效和名称。
输入： 无
输出： 保存到文件 f'all_stock_info_{datetime.now():%Y%m%d}.csv'

【参考资料】
1. 技术文档地址: https://ai-cyber-security.feishu.cn/docx/AB1fdqV9xo9K9axwT3hcMOOcnzc
2. BaoStock官网: http://baostock.com/baostock/index.php/Python_API文档

'''
import os, sys 
from datetime import datetime, timedelta
import pandas as pd
from utils.stock_funs import *

# 每个交易日的股票信息都可以单独下载
stock_date = '2025-01-03'
stock_code = 'sh.600710'
stock_file = f'./daily_data/{stock_code}.csv'

df = pd.read_csv(stock_file)

df.date = df.date.astype('datetime64[ns]')

df = df[(df['date'].dt.month == 9) & (df['date'].dt.day == 24)]

print(df.head())